how to create a menu in python